package com.feige.utils.file;

import com.feige.constant.Constants;

import java.util.Arrays;

/**
 * @author feige<br />
 * @ClassName: MyFileUtil <br/>
 * @Description: <br/>
 * @date: 2021/6/7 9:32<br/>
 */
public class MyFileUtil {


    /**
     * @description: 通过文件后缀获取对应的整型类型
     *文件类型定义：1（图片），2（音频），3（视频），4（xlsx或xls），5（ppt或pptx），
     *6（doc或docx），7（pdf），8（html），9（txt），10（md），11（zip或rar或jar）
     * @author: feige
     * @date: 2021/6/7 9:33
     * @param	suffix 文件后缀
     * @return: int
     */
    public static int getFileTypeByFileSuffix(String suffix){
        if (Arrays.asList(Constants.IMG_TYPE).contains(suffix)){
            return 1;
        }else if (Arrays.asList(Constants.VOICE_TYPE).contains(suffix)){
            return 2;
        }else if (Arrays.asList(Constants.AUDIO_TYPE).contains(suffix)){
            return 3;
        }else if (Arrays.asList(Constants.EXCEL_TYPE).contains(suffix)){
            return 4;
        }else if (Arrays.asList(Constants.PPT_TYPE).contains(suffix)){
            return 5;
        }else if (Arrays.asList(Constants.WORD_TYPE).contains(suffix)){
            return 6;
        }else if (Constants.PDF_TYPE.equals(suffix)){
            return 7;
        }else if (Constants.HTML_TYPE.equals(suffix)){
            return 8;
        }else if (Constants.TXT_TYPE.equals(suffix)){
            return 9;
        }else if (Constants.MD_TYPE.equals(suffix)){
            return 10;
        }else if (Arrays.asList(Constants.ZIP_TYPE).contains(suffix)) {
            return 6;
        }else {
            return 12;
        }

    }
}
